line fitting in python